# -*- coding: latin-1 -*-
from PyQt4.QtSql import QSqlRelationalTableModel, QSqlQuery
from PyQt4.QtCore import Qt, QVariant

DEBUG = False

class debe_modelo():
    def __init__(self):
        # Defino constantes que hacen referencia a las columnas. Es para llamarlas por nombre en vez de por numeros
        self.id = 0
        self.costo_tratamiento = 1 
        self.id_sinopsis =2
        self.id_duenio = 3
        
        # Se define el modelo
        self.model = QSqlRelationalTableModel()
        self.model.setTable("debe")       
        self.model.setHeaderData(self.costo_tratamiento, Qt.Horizontal, QVariant("Costo del tratamiento"))
        self.model.select()
        
    def eliminarRegistro(self, idSinopsis):
        self.query = QSqlQuery()
        if DEBUG: print "borrando registro de debe de la sinopsis: ", str(idSinopsis)
        self.query.exec_("delete from debe where id_sinopsis = "+str(idSinopsis)+";")
        self.model.select()
        
    def obtenerTotal(self, idDuenio):
        query = QSqlQuery("select sum(costotratamiento) from debe where id_duenio = "+str(idDuenio))
        if query.next():
            total = query.value(0).toDouble()[0];
        else:
            if DEBUG: print "La consulta no devolvio nada"
            total = -1.0
        return total
        
